# import os
#
# import dotenv
# from langchain_openai import ChatOpenAI
#
# # 加载环境变量
# dotenv.load_dotenv()
#
#
# def get_deepseek() -> ChatOpenAI:
#     """
#     获取DeepSeek客户端
#     :return: DeepSeek客户端
#     """
#     return ChatOpenAI(
#         model="deepseek-chat",  # 使用DeepSeek-V3 Chat Model
#         api_key=os.getenv("DEEPSEEK_API_KEY"),
#         base_url=os.getenv("DEEPSEEK_API_BASE"),
#     )


import os
from dotenv import load_dotenv
from langchain_openai import ChatOpenAI

# 加载环境变量（优先加载.env文件）
load_dotenv()


def get_deepseek() -> ChatOpenAI:
    """
    配置 DeepSeek 客户端
    """
    # 验证环境变量是否存在
    api_key = os.getenv("DEEPSEEK_API_KEY")
    api_base = os.getenv("DEEPSEEK_API_BASE")

    if not api_key:
        raise ValueError("请在.env文件或系统环境变量中设置 DEEPSEEK_API_KEY")
    if not api_base:
        api_base = "https://api.deepseek.com"  # 设置默认值

    return ChatOpenAI(
        model="deepseek-chat",  # 使用DeepSeek-V3模型
        api_key=api_key,
        base_url=api_base,
        temperature=0.7,  # 控制生成随机性（0-2）
        max_tokens=4096,  # 最大生成长度
    )


# 使用示例
if __name__ == "__main__":
    client = get_deepseek()
    response = client.invoke(
        "用Python实现快速排序算法"
    )
    print(response.content)